// Copyright (c) 2025, 大连理工大学 (Dalian University of Technology)
//
// Licensed under the Mulan PSL v2.
// You can use this file according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of the License at
//
//     http://license.coscl.org.cn/MulanPSL2
//
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
// EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. SEE THE MULAN PSL v2
// FOR MORE DETAILS.

/**
 * @brief
 *
 * @version  0.1
 * @author   wuxingchuang
 * @date     2025.06.26
 * @copyright Copyright (c) 2025
 */

#include "gtest/gtest.h"

#include "Utility/FileSystem/SGFileSystem.h"
#include "Utility/Logging/SGLogger.h"

#include "DBManager/DBServiceFactory.h"
#include "DBManager/IService/Input/Parameter/ISETSService.h"

#include "Import/BDFImport/BDFImport.h"
#include "Import/BDFImport/BDFParserBase.h"

using namespace SG::DataStructure::FEM;
using namespace SG::DBManager;
using namespace SG::Import;

extern std::shared_ptr<DBServiceFactory> DB_SERVICEFACTORY;
extern std::vector<std::string>          dividedEntryInfo;

TEST (SETTest, test1)
{
    DB_SERVICEFACTORY->ClearDatabase ();

    auto _imp     = std::make_shared<BDFImporter> (DB_SERVICEFACTORY);
    auto filePath = SG::Utility::FileSystem::getTestDataPath ("Import/BDF/trcompc.bdf");
    _imp->Load (filePath.c_str ());

    auto setService = DB_SERVICEFACTORY->get<ISETSService> ();

    std::vector<SG::DataStructure::FEM::SETSData> sets;
    ASSERT_EQ (setService->FindBySId (10, sets), SG::DataStructure::Common::Status::STATUS_Success);
    ASSERT_EQ (sets.size (), 4);

    SG::DataStructure::FEM::SETSData set;
    ASSERT_EQ (setService->Find (20, 503, set), SG::DataStructure::Common::Status::STATUS_Success);
    ASSERT_EQ (set.m_memeIds.size (), 112);
    ASSERT_EQ (set.m_memeIds.back (), 3070);

    ASSERT_EQ (setService->Find (20, 1, set), SG::DataStructure::Common::Status::STATUS_Success);
    ASSERT_EQ (set.m_memeIds.size (), 4);

    ASSERT_EQ (setService->Find (20, 2, set), SG::DataStructure::Common::Status::STATUS_Success);
    ASSERT_EQ (set.m_memeIds.size (), 3);
}
